# coding: UTF-8
import matplotlib.pyplot as plt


# x_values = [1, 2, 3, 4, 5]
# y_values = [1, 4, 9, 16, 25]

x_values = list(range(1, 1001))
y_values = [x**2 for x in x_values]

# 设置颜色
# color或c表示颜色；edgecolor表示圆点边缘
point_color = "red"     # 用关键词设定颜色
point_color = (0, 0.6, 1)  # 用RGB设定颜色（每个颜色的分量在0~1之间）
point_color = y_values  # 配合cmap参数，实现颜色映射（此时颜色参数必须用c，而非color）
plt.scatter(x_values, y_values, c=point_color, cmap=plt.cm.Blues, edgecolor="none", s=40)    # 参数：s 表示点的大小


# 设置图表标题、x轴标签、y轴标签
plt.title("Square Numbers", fontsize=24)
plt.xlabel("Value", fontsize=14)
plt.ylabel("Square of Value", fontsize=14)

# 设置刻度标记的大小
plt.tick_params(axis="both", which="major", labelsize=14)

# 设置每个坐标轴的取值范围
plt.axis([0, 1100, 0, 1100000])

# 保存绘制的图形
plt.savefig("squares_plot.png", bbox_inches="tight")    # 第二个实参指定将图表多余的空白区域裁剪掉

plt.show()
